Papers by Adithya M Devraj
AutoMixAlign: Adaptive Data Mixing for Multi-Task Preference Optimization in LLMs (2025.acl-long)
Copied to clipboard
Nicholas E. Corrado, Julian Katz-Samuels, Adithya M Devraj, Hyokun Yun, Chao Zhang, Yi Xu, Yi Pan, Bing Yin, Trishul Chilimbi
| Challenge: | Existing approaches to align large language models rely on large ablation studies, heuristics, or human intuition to produce models with strong performance across tasks. |
| Approach: | They propose an algorithm that mixes datasets during LLM training to balance performance across multiple tasks. |
| Outcome: | The proposed algorithm outperforms existing methods on multitask alignment setups and achieves convergence rate of O(1/T) in the convex case. |